check the memory usage in python